/*
 *
 *  * | Licensed 未经许可不能去掉「OPENIITA」相关版权
 *  * +----------------------------------------------------------------------
 *  * | Author: xw2sy@163.com
 *  * +----------------------------------------------------------------------
 *
 *  Copyright [2024] [OPENIITA]
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 * /
 */

package cc.iotkit.data.service;

import cc.iotkit.common.api.Paging;
import cc.iotkit.common.satoken.utils.AuthUtil;
import cc.iotkit.common.satoken.utils.LoginHelper;
import cc.iotkit.common.utils.MapstructUtils;
import cc.iotkit.data.dao.DeviceGroupMappingRepository;
import cc.iotkit.data.dao.DeviceGroupRepository;
import cc.iotkit.data.dao.IJPACommData;
import cc.iotkit.data.manager.IDeviceGroupData;
import cc.iotkit.data.model.TbDeviceGroup;
import cc.iotkit.data.model.TbDeviceInfo;
import cc.iotkit.model.device.DeviceGroup;
import cc.iotkit.model.device.DeviceInfo;
import cn.hutool.core.collection.CollUtil;
import com.github.yitter.idgen.YitIdHelper;
import com.querydsl.jpa.impl.JPAQuery;
import com.querydsl.jpa.impl.JPAQueryFactory;
import lombok.RequiredArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Primary;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import static cc.iotkit.data.model.QTbDeviceGroup.tbDeviceGroup;
import static cc.iotkit.data.model.QTbDeviceGroupMapping.tbDeviceGroupMapping;
import static cc.iotkit.data.model.QTbDeviceInfo.tbDeviceInfo;

@Primary
@Service
@RequiredArgsConstructor
public class DeviceGroupDataImpl implements IDeviceGroupData, IJPACommData<DeviceGroup, String> {

    @Autowired
    private DeviceGroupRepository deviceGroupRepository;

    private final DeviceGroupMappingRepository deviceGroupMappingRepository;

    private final JPAQueryFactory jpaQueryFactory;

    @Override
    public JpaRepository getBaseRepository() {
        return deviceGroupRepository;
    }

    @Override
    public Class getJpaRepositoryClass() {
        return TbDeviceGroup.class;
    }

    @Override
    public Class getTClass() {
        return DeviceGroup.class;
    }

    @Override
    public Paging<DeviceGroup> findByNameLike(String groupId, String name, int page, int size) {
        /*Page<TbDeviceGroup> groups = deviceGroupRepository.findByNameLike("%" + name.trim() + "%",
                Pageable.ofSize(size).withPage(page - 1));
        return new Paging<>(groups.getTotalElements(),
                MapstructUtils.convert(groups.getContent(), DeviceGroup.class));*/

        JPAQuery<TbDeviceGroup> query = jpaQueryFactory.selectFrom(tbDeviceGroup);
        if (!LoginHelper.isSuperAdmin()) {
            query.where(tbDeviceGroup.uid.eq(AuthUtil.getUserId()));
        }
        if (StringUtils.isNotBlank(groupId)) {
            query.where(tbDeviceGroup.id.eq(groupId));
        }
        if (StringUtils.isNotBlank(name)) {
            query.where(tbDeviceGroup.name.like("%" + name.trim() + "%"));
        }

        query.offset((page - 1) * size).limit(size);

        List<TbDeviceGroup> tbDeviceGroups = query.fetch();
        long total = query.fetchCount();
        return new Paging<>(total, MapstructUtils.convert(tbDeviceGroups, DeviceGroup.class));
    }

    /**
     * 根据用户查询分组
     * @param page,siaze
     * @return
     */
    @Override
    public Paging<DeviceGroup> findByUserid(int page, int size) {
        Page<TbDeviceGroup> groups ;
        if (AuthUtil.isAdmin()){
            groups = deviceGroupRepository.findByUserid(Pageable.ofSize(size).withPage(page - 1));

        }else {
            groups = deviceGroupRepository.findByUserid(AuthUtil.getUserId(),Pageable.ofSize(size).withPage(page - 1));
        }
        return new Paging<>(groups.getTotalElements(),
                MapstructUtils.convert(groups.getContent(), DeviceGroup.class));
    }

    @Override
    public DeviceGroup findByGroupCode(String groupCode) {
        TbDeviceGroup group = deviceGroupRepository.findByGroupCodeAndUid(groupCode, AuthUtil.getUserId());
        return MapstructUtils.convert(group, DeviceGroup.class);
    }

    @Override
    public List<DeviceInfo> findDevicesByGroupId(String groupId) {
        JPAQuery<TbDeviceInfo> query = jpaQueryFactory.selectFrom(tbDeviceInfo);
        query.join(tbDeviceGroupMapping).on(tbDeviceGroupMapping.deviceId.eq(tbDeviceInfo.deviceId));
        query.where(tbDeviceGroupMapping.groupId.eq(groupId));
        List<TbDeviceInfo> tbDeviceInfos = query.fetch();

        List<DeviceInfo> deviceInfos = new ArrayList<>(tbDeviceInfos.size());
        for (TbDeviceInfo tbDeviceInfo : tbDeviceInfos) {
            DeviceInfo deviceInfo = MapstructUtils.convert(tbDeviceInfo, DeviceInfo.class);
            parseStateToDto(tbDeviceInfo, deviceInfo);
            deviceInfos.add(deviceInfo);
        }
        return deviceInfos;
    }

    /**
     * 将设备状态从vo转为dto的
     */
    private void parseStateToDto(TbDeviceInfo vo, DeviceInfo dto) {
        dto.setState(new DeviceInfo.State("online".equals(vo.getState()),
                vo.getOnlineTime(), vo.getOfflineTime()));
        dto.setLocate(new DeviceInfo.Locate(vo.getLongitude(), vo.getLatitude()));
    }

    @Override
    public DeviceInfo findFirstDeviceByGroupId(String groupId) {
        JPAQuery<TbDeviceInfo> query = jpaQueryFactory.selectFrom(tbDeviceInfo);
        query.join(tbDeviceGroupMapping).on(tbDeviceGroupMapping.deviceId.eq(tbDeviceInfo.deviceId));
        query.where(tbDeviceGroupMapping.groupId.eq(groupId));
        List<TbDeviceInfo> tbDeviceInfos = query.fetch();
        return CollUtil.isEmpty(tbDeviceInfos) ? null : MapstructUtils.convert(tbDeviceInfos.get(0), DeviceInfo.class);
    }

    @Override
    public List<String> findGroupConfigByDeviceId(String deviceId) {
        JPAQuery<TbDeviceGroup> query = jpaQueryFactory.selectFrom(tbDeviceGroup);
        query.join(tbDeviceGroupMapping).on(tbDeviceGroupMapping.groupId.eq(tbDeviceGroup.id));
        query.where(tbDeviceGroupMapping.deviceId.eq(deviceId));
        List<TbDeviceGroup> tbDeviceGroups = query.fetch();
        return CollUtil.isEmpty(tbDeviceGroups) ? CollUtil.newArrayList() : tbDeviceGroups.stream().map(tbDeviceGroup ->
                tbDeviceGroup.getGroupCode() + "_" + tbDeviceGroup.getUid()).collect(Collectors.toList());
    }

    @Override
    public String findGroupConfigByGroupId(String groupId) {
        TbDeviceGroup tbDeviceGroup = deviceGroupRepository.findById(groupId).orElse(null);
        return tbDeviceGroup == null ? "" : tbDeviceGroup.getGroupCode() + "_" + tbDeviceGroup.getUid();
    }

    @Override
    public DeviceGroup findById(String s) {
        return MapstructUtils.convert(deviceGroupRepository.findById(s).orElse(null), DeviceGroup.class);
    }

    @Override
    public DeviceGroup save(DeviceGroup data) {
        if (StringUtils.isBlank(data.getId())) {
            // data.setId(UUID.randomUUID().toString());
            data.setId(String.valueOf(YitIdHelper.nextId()));
        }
        deviceGroupRepository.save(MapstructUtils.convert(data, TbDeviceGroup.class));
        return data;
    }


    @Override
    public List<DeviceGroup> findAll() {
        return MapstructUtils.convert(deviceGroupRepository.findAll(), DeviceGroup.class);
    }


}
